import pymysql
from pymysql.cursors import DictCursor



# 连接数据库，用函数封装起来，方便后续使用
def get_conn():
    return pymysql.connect(
        host='127.0.0.1',
        user='root',
        password='123456',
        database='python_mysql',    # 在数据库里新建的数据库的名字
        charset='utf8'
)









# 查询数据库的数据
def query_data(sql):    # 这里的参数sql是即将产生业务的sql语句
    conn = get_conn()
    try:
        cursor = conn.cursor(DictCursor)
        cursor.execute(sql)
        return cursor.fetchall()
    finally:
        conn.close()






# 新增和更新数据
def insert_or_update_data(sql):    # 这里的参数sql是即将产生业务的sql语句
    conn = get_conn()
    try:
        cursor = conn.cursor(DictCursor)
        cursor.execute(sql)
        conn.commit()  # 提交更新
    finally:
        conn.close()







